from langchain.prompts import ChatPromptTemplate
from tool import langchain_client
if __name__ == '__main__':
    # 首先，构造一个提示模版字符串:`template_string`
    template_string = """把由三个反引号分隔的文本` 翻译成一种{style}风格。
    文本: ```{text}```
    """
    # 然后，我们调用`ChatPromptTemplatee.from_template()`函数将
    # 上面的提示模版字符`template_string`转换为提示模版`prompt_template`
    prompt_template = ChatPromptTemplate.from_template(template_string)

    service_reply = """嘿，顾客，\
    保修不包括厨房的清洁费用， \
    因为您在启动搅拌机之前 \
    忘记盖上盖子而误用搅拌机, \
    这是您的错。 \
    倒霉! 再见!
    """
    service_style_pirate = """\
    一个有礼貌的语气 \
    使用海盗风格 \
    """
    service_messages = prompt_template.format_messages(
        style=service_style_pirate,
        text=service_reply)
    print("\n", service_messages[0].content)

    # 调用模型部分定义的chat模型来转换回复消息风格
    service_response = langchain_client.invoke(service_messages)
    print(service_response.content)

